

import os

os.environ["ARK_API_KEY"] = '' #你的ARK API Key
os.environ["BASE_URL"] = 'https://ark.cn-beijing.volces.com/api/v3' #OpenAI 的 API URL
os.environ["LLM_MODELEND"] = 'ep-20241104131149-csxf9' #你的模型ID

from langchain_openai import ChatOpenAI
from langchain.schema import HumanMessage, SystemMessage

chat = ChatOpenAI(
    model=os.environ.get("LLM_MODELEND"), 
    api_key=os.environ.get("ARK_API_KEY"),
    base_url=os.environ.get("BASE_URL"),
    temperature=0.8, max_tokens=600
)

messages = [
    SystemMessage(content="你是一个很棒的智能助手"),
    HumanMessage(content="请给我的花店起个名"),
]
response = chat(messages)
print(response)


